/*
 *  Copyright 1999-2019 Seata.io Group.
 *
 *  Licensed under the Apache License, Version 2.0 (the "License");
 *  you may not use this file except in compliance with the License.
 *  You may obtain a copy of the License at
 *
 *       http://www.apache.org/licenses/LICENSE-2.0
 *
 *  Unless required by applicable law or agreed to in writing, software
 *  distributed under the License is distributed on an "AS IS" BASIS,
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  See the License for the specific language governing permissions and
 *  limitations under the License.
 */
package io.seata.server.session;

import java.util.Collection;

import io.seata.core.context.RootContext;
import io.seata.core.exception.TransactionException;
import io.seata.core.model.BranchType;
import io.seata.core.model.GlobalStatus;
import io.seata.server.UUIDGenerator;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.slf4j.MDC;

/**
 * The type Session helper.
 *
 * @author sharajava
 */
public class SessionHelper {

	private static final Logger LOGGER = LoggerFactory.getLogger(SessionHelper.class);

	private SessionHelper() {
	}

	public static BranchSession newBranchByGlobal(GlobalSession globalSession, BranchType branchType, String resourceId,
			String lockKeys, String clientId) {
		return newBranchByGlobal(globalSession, branchType, resourceId, null, lockKeys, clientId);
	}

	/**
	 * New branch by global branch session.
	 * @param globalSession the global session
	 * @param branchType the branch type
	 * @param resourceId the resource id
	 * @param lockKeys the lock keys
	 * @param clientId the client id
	 * @return the branch session
	 */
	public static BranchSession newBranchByGlobal(GlobalSession globalSession, BranchType branchType, String resourceId,
			String applicationData, String lockKeys, String clientId) {
		BranchSession branchSession = new BranchSession();

		branchSession.setXid(globalSession.getXid());
		branchSession.setTransactionId(globalSession.getTransactionId());
		branchSession.setBranchId(UUIDGenerator.generateUUID());
		branchSession.setBranchType(branchType);
		branchSession.setResourceId(resourceId);
		branchSession.setLockKey(lockKeys);
		branchSession.setClientId(clientId);
		branchSession.setApplicationData(applicationData);

		return branchSession;
	}

	/**
	 * New branch
	 * @param branchType the branch type
	 * @param xid Transaction id.
	 * @param branchId Branch id.
	 * @param resourceId Resource id.
	 * @param applicationData Application data bind with this branch.
	 * @return the branch session
	 */
	public static BranchSession newBranch(BranchType branchType, String xid, long branchId, String resourceId,
			String applicationData) {
		BranchSession branchSession = new BranchSession();
		branchSession.setXid(xid);
		branchSession.setBranchId(branchId);
		branchSession.setBranchType(branchType);
		branchSession.setResourceId(resourceId);
		branchSession.setApplicationData(applicationData);
		return branchSession;
	}

	/**
	 * End committed.
	 * @param globalSession the global session
	 * @throws TransactionException the transaction exception
	 */
	public static void endCommitted(GlobalSession globalSession) throws TransactionException {
		globalSession.changeStatus(GlobalStatus.Committed);
		globalSession.end();
	}

	/**
	 * End commit failed.
	 * @param globalSession the global session
	 * @throws TransactionException the transaction exception
	 */
	public static void endCommitFailed(GlobalSession globalSession) throws TransactionException {
		globalSession.changeStatus(GlobalStatus.CommitFailed);
		globalSession.end();
	}

	/**
	 * End rollbacked.
	 * @param globalSession the global session
	 * @throws TransactionException the transaction exception
	 */
	public static void endRollbacked(GlobalSession globalSession) throws TransactionException {
		GlobalStatus currentStatus = globalSession.getStatus();
		if (isTimeoutGlobalStatus(currentStatus)) {
			globalSession.changeStatus(GlobalStatus.TimeoutRollbacked);
		}
		else {
			globalSession.changeStatus(GlobalStatus.Rollbacked);
		}
		globalSession.end();
	}

	/**
	 * End rollback failed.
	 * @param globalSession the global session
	 * @throws TransactionException the transaction exception
	 */
	public static void endRollbackFailed(GlobalSession globalSession) throws TransactionException {
		GlobalStatus currentStatus = globalSession.getStatus();
		if (isTimeoutGlobalStatus(currentStatus)) {
			globalSession.changeStatus(GlobalStatus.TimeoutRollbackFailed);
		}
		else {
			globalSession.changeStatus(GlobalStatus.RollbackFailed);
		}
		globalSession.end();
	}

	public static boolean isTimeoutGlobalStatus(GlobalStatus status) {
		return status == GlobalStatus.TimeoutRollbacked || status == GlobalStatus.TimeoutRollbackFailed
				|| status == GlobalStatus.TimeoutRollbacking || status == GlobalStatus.TimeoutRollbackRetrying;
	}

	/**
	 * Foreach global sessions.
	 * @param sessions the global sessions
	 * @param handler the handler
	 * @since 1.5.0
	 */
	public static void forEach(Collection<GlobalSession> sessions, GlobalSessionHandler handler) {
		for (GlobalSession globalSession : sessions) {
			try {
				MDC.put(RootContext.MDC_KEY_XID, globalSession.getXid());
				handler.handle(globalSession);
			}
			catch (Throwable th) {
				LOGGER.error("handle global session failed: {}", globalSession.getXid(), th);
			}
			finally {
				MDC.remove(RootContext.MDC_KEY_XID);
			}
		}
	}

	/**
	 * Foreach branch sessions.
	 * @param sessions the branch session
	 * @param handler the handler
	 * @since 1.5.0
	 */
	public static Boolean forEach(Collection<BranchSession> sessions, BranchSessionHandler handler)
			throws TransactionException {
		Boolean result;
		for (BranchSession branchSession : sessions) {
			try {
				MDC.put(RootContext.MDC_KEY_BRANCH_ID, String.valueOf(branchSession.getBranchId()));
				result = handler.handle(branchSession);
				if (result == null) {
					continue;
				}
				return result;
			}
			finally {
				MDC.remove(RootContext.MDC_KEY_BRANCH_ID);
			}
		}
		return null;
	}

}
